package com.sparsearr.datastructures.ztr.recursion;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

/**
 * @Author: ZhaoTR
 * @Date: Created in 2025/5/16 20:34
 * @Description: 八皇后
 * @Version: 1.0
 */

public class Queue8 {
    private static final Logger logger = LogManager.getLogger(Queue8.class);

    // 定义一个 max 表示共有多少个皇后
    int max = 8;

    // 定义数组 array 存放皇后放置位置的结果，比如 array[0] = 2 表示第0个皇后放在第2列
    int[] array = new int[max];

    // 记录一共有多少种解法
    static int count = 0;

    // 记录判断冲突的次数
    static int judgeCount = 0;

    public static void main(String[] args) {
        // 查看八皇后是否正确
        Queue8 queue8 = new Queue8();
        queue8.check(0);
        logger.info("一共有" + count + "种解法");
        logger.info("判断冲突的次数为" + judgeCount);
    }

    // 放置第n个皇后  check 是每一次递归时，进入到check中都有一次for循环，因此会有回溯
    private void check(int n) {
        // n = 8，其实前8个皇后就已经放好
        if (n == max) {
            print();
            return;
        }
        // 依次放入皇后，并判断是否冲突
        for (int i = 0; i < max; i++) {
            // 先把当前的皇后，放到该行的第一列
            array[n] = i;
            // 判断：当前的这个皇后n到第i列时，是否冲突
            if (judge(n)) {
                // 接着放置第n+1个皇后，即继续递归
                check(n + 1);
            }
            // 如果冲突，则继续尝试其他位置 array[n] = i; 即将第n个皇后，放置在本行的 后移一个位置
        }
    }

    // 查看当我们放置第n个皇后时，去检测该皇后和前面已经摆放的皇后是否合法

    /**
     * @param n 表示第n个皇后
     */
    private boolean judge(int n) {
        judgeCount ++ ;
        for (int i = 0; i < n; i++) {
            // 1. array[i] == array[n] 判断第n个皇后和第i个皇后是否在同一列
            // 2. Math.abs(n - 1) == Math.abs(array[n] - array[i]) 判断第n个皇后和第i个皇后是否在同一斜线
            // 3. 判断是否在同一行，每次n都在递增
            if (array[i] == array[n] || Math.abs(n - i) == Math.abs(array[n] - array[i])) {
                return false;
            }
        }
        return true;
    }

    // 将皇后摆放的位置打印出来
    private void print() {
        count ++ ;
        for (int j : array) {
            System.out.print(j + " ");
        }
        System.out.println();
    }
}
